python main_sbi.py --batch_size 30 \
                    --steps_for_eval 100 \
                    --steps_eval -1 \
                    --resume true \
                    --resume_epoch 9 \
                    --model_dir ./model_2 \
                    --mode test